import os

# 设置 PROJECT_ROOT 为 config.py 所在的目录
os.environ['PROJECT_ROOT'] = os.path.dirname(os.path.abspath(__file__))
BASE_DIR = os.path.abspath(os.path.dirname(__file__))

scale_factor = 3 # 截取人脸的时候,需要放大的倍数

models_directory = os.path.join(os.environ['PROJECT_ROOT'], 'models')

notify_token = "inUVBuypkgdmm5ym6l3XiIMwenb7SKQu"
notify_url  = "http://api.gpuhub.xiaolubi.com/task/queue/completion"
similarity_threshold = 0.45  # 多人换脸的人脸阈值

qv = 5  # 图片切割的质量
crf = 18 # 最终合成视频的压缩程度,越大的话视频越不清晰

import os
# 这个路径就先暂时这样吧,先放在一起
kLocUserImagesPath = "/root/autodl-tmp/image_data/users_images"
kSourceVideosFrame = "/root/autodl-tmp/image_data/source_videos_frame"
kSourceGIFFrame = "/root/autodl-tmp/image_data/source_gif_frame"
kUserSwapperFaceFrames = "/root/autodl-tmp/image_data/user_swapper_face_frames"
kSourceVideosData = "/root/autodl-tmp/image_data/source_videos_data"  # 这个似乎没有

folders = [kLocUserImagesPath, kSourceVideosFrame, kSourceGIFFrame, kUserSwapperFaceFrames, kSourceVideosData]
# 遍历文件夹路径，检查每个文件夹是否存在，如果不存在，就创建它
for folder in folders:
    if not os.path.exists(folder):
        os.makedirs(folder)

# 这几个应该都用不到了
kSwapperModels = os.path.join(BASE_DIR, "models/inswapper_128.onnx")
kFaceEnhancerModels = os.path.join(BASE_DIR, "models/GFPGANv1.4.pth")

kSlideOutputPath = os.path.join(BASE_DIR, "Plugin/图片合成幻灯片/slide_output")

# 字体文件的路径
kFontPath = os.path.join(BASE_DIR, "resources/LXGWHeartSerifCHS.ttf")


# 本地的redis还是要的,因为需要做通知用
redis_config = {
    'host': 'localhost',
    'port': 6379,
    'decode_responses': False  # 或者 False，根据你的需求
}
